import torch
from torchsummary import summary
from nets.get_net import getNet

model = getNet('SwinUnetwDAmoduleV1').cuda()
summary(model, (4, 160, 160))
